Python: Average of items in matrix